from dlp_gui import MyGUI
from dlp_paper_downloader import Ieee2
from dlp_pdf_parser import TestParsePdfReferences
from dlp_pdf_summary import PDFSummary

my_keyword=''

class MyGUI(MyGUI):
    def __init__(self):
        super().__init__()
    def on_submit(self):
        global my_url
        print("Submitting...")
        url = self.url_entry.get()
        if url:
            print(f"输入的关键词: {url}")
            my_keyword = url
            self.app.destroy()
            # 在这里添加你的处理逻辑，例如验证链接或发送请求
        else:
            print("请输入一个关键词")


if __name__ == "__main__":
    # GUI界面
    mygui=MyGUI()
    print(my_keyword)
    # 下载IEEE论文
    ieee=Ieee2(key=my_keyword)
    ieee.main()
    # 解析PDF生成引用
    test1=TestParsePdfReferences()
    test1.test_parse_pdf_references()
    # 用T5模型生成摘要
    pdfsummary=PDFSummary()
    pdfsummary.pdf_summary()